package utils

import (
	"github.com/go-xorm/xorm"
)

//Transaction xorm处理事务时 错误必须手动rollback
//简单封装下 避免手动rollback
func Transaction(e *xorm.Engine, closure func(*xorm.Session) error) error {
	si := e.NewSession()
	defer si.Close()

	err := si.Begin()
	if err != nil {
		return err
	}
	err = func() (e error) {
		defer func() {
			if err := recover(); err != nil {
				//e = fmt.Errorf("%v", err)//如果panic 转化为error字符串
				si.Rollback()
				panic(err)
			}
		}()
		return closure(si)
	}()

	if err != nil {
		si.Rollback()
		return err
	}

	return si.Commit()
}
